def normlize(param):
    return param.norm(ord=2, axis=1, keepdims=True)
